import requests

def test():
    url = 'http://127.0.0.1:8080/api/v0/generate'
    req_data = dict(
        prompt='"hello, how are you',
        num_return_sequences=2,
        stop_words = ["I'm", "good"]
    )
    rsp = requests.post(url, json=req_data)
    if rsp.status_code ==200:
        rsp_data = rsp.json()
        print(rsp_data)
    else:
        print(rsp.status_code)

test()
